_base_ = ["./voc0712.py", "./schedule_2x.py", "./default_runtime.py"]
# model setting
backbone_pretrained_path = (
    "https://github.com/d-li14/mobilenetv2.pytorch/blob"
    "/master/pretrained/mobilenetv2-c5e733a8.pth")
model = dict(type="CenterNet",
             pretrained=backbone_pretrained_path,
             backbone=dict(type="MobileNetV2",
                           width_mult=1.0,
                           out_feature_names=["3", "6", "13", "conv"]),
             neck=dict(type="LayerAggregation",
                       in_channels_list=[24, 32, 96, 1280],
                       out_channels=256),
             bbox_head=dict(type="CenterHead",
                            num_classes=20,
                            in_channels=256,
                            loss_heatmap=dict(type="GaussianFocalLoss",
                                              alpha=2.0,
                                              gamma=4.0,
                                              loss_weight=1),
                            loss_wh_size=dict(type="L1Loss",
                                              reduction="mean",
                                              loss_weight=0.1),
                            loss_ct_off=dict(type="L1Loss",
                                             reduction="mean",
                                             loss_weight=1)))
train_cfg = None
test_cfg = None
